package com.tianlan.blog.shiro;

import com.tianlan.blog.entity.SysUser;
import com.tianlan.common.annotation.auth.CurrentUser;
import com.tianlan.common.exception.UnauthorizedException;

import org.springframework.core.MethodParameter;
import org.springframework.web.bind.support.WebDataBinderFactory;
import org.springframework.web.context.request.NativeWebRequest;
import org.springframework.web.context.request.RequestAttributes;
import org.springframework.web.method.support.HandlerMethodArgumentResolver;
import org.springframework.web.method.support.ModelAndViewContainer;

/**
 * @ClassName: CurrentUserMethodArgumentResolver
 * @Description: 增加方法注入，将含有 @CurrentUser 注解的方法参数注入当前登录用户
 * @author 
 * @date 
 */
public class CurrentUserMethodArgumentResolver implements HandlerMethodArgumentResolver {

    @Override
    public boolean supportsParameter(MethodParameter parameter) {
        return parameter.getParameterType().isAssignableFrom(SysUser.class)
                && parameter.hasParameterAnnotation(CurrentUser.class);
    }

    @Override
    public Object resolveArgument(MethodParameter parameter, ModelAndViewContainer mavContainer,
            NativeWebRequest webRequest, WebDataBinderFactory binderFactory) throws Exception {
    	SysUser sysUser = (SysUser) webRequest.getAttribute("currentUser", RequestAttributes.SCOPE_REQUEST);
        if (sysUser == null) {
            throw new UnauthorizedException("获取用户信息失败");
        }
        return sysUser;
    }
}
